#pragma once
#include <iostream>

#include "Socket.hpp"
#include "Common.hpp"
#include "Reactor.hpp"
#include "Connection.hpp"
using namespace socket_ns;

// 用来统一管理Listen套接字
class Listener
{
public:
    Listener(uint16_t port) : _port(port), _listensock(std::make_unique<TcpSocket>())
    {
        _listensock->BuildListenSocket(_port);
    }
    int ListenSockfd()
    {
        return _listensock->Sockfd();
    }
    void Accepter(Connection *conn)
    {
        // 由于我们采用的是ET模式，循环获取底层连接（一次性获取完），非阻塞
        // 但是这里有可能只有一个listen套接字
        while (true)
        {
            // 每次循环时，都将errno清0
            errno = 0;
            InetAddr addr;
            int code = 0;
            int sockfd = _listensock->Accepter(&addr, &code);
            if (sockfd > 0)
            {
                // 获取连接成功
                LOG(INFO, "获取连接成功， 客户端信息：%s:%d, new_sockfd: %d\n", addr.Ip().c_str(), addr.Port(), sockfd);
                conn->_R->AddConnection(sockfd, EPOLLIN | EPOLLET, addr, NormalConnection); // 需要添加对普通套接字进行IO事件处理的模块
            }
            else
            {
                // 失败
                if (code == EWOULDBLOCK)
                {
                    LOG(INFO, "底层连接全部获取完毕\n");
                    break;
                }
                else if (code == EINTR)
                {
                    continue;
                }
                else
                {
                    LOG(ERROR, "获取连接失败！\n");
                    break;
                }
            }
        }
    }
    ~Listener() {}

private:
    std::unique_ptr<Socket> _listensock;
    uint16_t _port;
};
